config = {
    'device': 'cuda',  # 'cpu' or 'cuda'
    'root': r'C:\Users\admin\Desktop\与乔亲成同学的合作\OCTA_DPN_Diagnose\data\batch_2\clean',  # 数据集的根目录
    'val_frac': 0.3,  # 验证集在整个数据集当中的比例
    'shuffle': True,  # 是否将数据集打乱
    'train_batch_size': 16,  # 训练集的batch大小
    'val_batch_size': 16,
    # 'train_batch_size': 2048,  # 训练集的batch大小
    # 'val_batch_size': 2048,
    'epoch_num': 600,
    "initial_lr": 0.000001,  # 初始学习率
    "max_lr": 0.0003,  # 最大学习率（在warm-up结束时达到）
    "warmup_epochs": 8,
    'weight_decay': 0,
    'data_types': [  # 筛选使用每个病人的哪些类型的数据
        '表层血流.jpg',
        '脉络膜毛细血管.jpg',
        '深层血流.jpg',
        '视网膜血流.jpg',
        '玻璃体血流.jpg',
        '无血管层血流.jpg',
    ],
    'save': True,
}

